import torch

path = r"../data/cmn.txt"  # data path
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
d_model = 128
dtype = torch.long  # encoder input type
batch_size = 1
max_len = 16
teacher_force = 0.8
debuging = 0  # debug mode open = True
lr_en = 1e-4  # encoder learning rate
lr_cn = 1e-4
epochs = 100
checkout_point = 10  # checkout frequency
verbose_point = 1  # verbose fre
is_load_model = True # load model from local
split_ratio = 0.8
momentum=0.8